/*
 * Copyright (C) 2018 The Asiainfo-Ability Authors
 *
 *      Licensed under the Apache License, Version 2.0 (the "License");
 *      you may not use this file except in compliance with the License.
 *      You may obtain a copy of the License at
 *
 *          http://www.apache.org/licenses/LICENSE-2.0
 *
 *      Unless required by applicable law or agreed to in writing, software
 *      distributed under the License is distributed on an "AS IS" BASIS,
 *      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *      See the License for the specific language governing permissions and
 *      limitations under the License.
 */

package org.asiainfo.ability.gateway.custom;

import com.netflix.zuul.context.RequestContext;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.HttpUriRequest;
import org.asiainfo.ability.base.vo.Record;
import org.asiainfo.ability.gateway.context.AbilityRequestContext;
import org.springframework.cloud.netflix.ribbon.apache.RibbonApacheHttpRequest;
import org.springframework.cloud.netflix.zuul.filters.route.RibbonCommandContext;

/**
 * Customize the RibbonApacheHttpRequest for dynamic setting ribbbon timeout
 *
 * @Author: visen
 * @Date: 2018/1/16
 * @Description:
 */
public class CustomRibbonApacheHttpRequest extends RibbonApacheHttpRequest {
    public CustomRibbonApacheHttpRequest(RibbonCommandContext context) {
        super(context);
    }

    @Override
    public HttpUriRequest toRequest(final RequestConfig requestConfig) {
        //dynamic

        AbilityRequestContext requestContext = (AbilityRequestContext) RequestContext.getCurrentContext()
                .get(AbilityRequestContext.CONTEXT_KEY);
        if (requestContext.getTimeout() > 0) {
            RequestConfig.Builder builder = RequestConfig.copy(requestConfig);
            builder.setSocketTimeout(requestContext.getTimeout());
            return super.toRequest(builder.build());
        }
        return super.toRequest(requestConfig);

    }
}
